cd ../../

gpus=$1

python sed_main.py \
    --gpus ${gpus}, \
    --exp_name finetune/OOD_Bsz512_Mask0.65Posemd_MixUpT_Crop1.01.0_FW0.61.0 \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/self/OOD_Bsz512_Mask0.65Posemd_MixUpT_Crop1.01.0_FW0.61.0/version_0/epoch=166-obj_metric=0.280.ckpt" \
    --pretrained_prefix "encoder_tea.encoder." \
    --desirable_prefix "" \
    --lrs '[0.001, 0.001, 0.001]' \
    --verbose True



    # --pretrained_ckpt '/home/shaonian/SED/sssl_sed/codes/exp/self/BYOLA_OOD/version_0/epoch=113-obj_metric=0.209.ckpt' \
    # --pretrained_prefix 'encoder_tea.cnn' \